package com.yonghui.springsecurityoauth2demo.pojo;


import cn.hutool.json.JSON;
import cn.hutool.json.JSONUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.HttpStatus;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.HashMap;

/**
 * 操作消息提醒
 *
 * @author admin
 */
public class AjaxResult extends HashMap<String, Object> {
    private static final Logger LOGGER = LoggerFactory.getLogger(AjaxResult.class);

    private static final long serialVersionUID = 1L;

    /**
     * 状态码
     */
    public static final String CODE_TAG = "code";

    /**
     * 返回内容
     */
    public static final String MSG_TAG = "msg";

    /**
     * 数据对象
     */
    public static final String DATA_TAG = "data";

    /**
     * 初始化一个新创建的 AjaxResult 对象，使其表示一个空消息。
     */
    public AjaxResult() {
    }

    /**
     * 初始化一个新创建的 AjaxResult 对象
     *
     * @param code 状态码
     * @param msg  返回内容
     */
    public AjaxResult(int code, String msg) {
        super.put(CODE_TAG, code);
        super.put(MSG_TAG, msg);
    }

    /**
     * 初始化一个新创建的 AjaxResult 对象
     *
     * @param code 状态码
     * @param msg  返回内容
     * @param data 数据对象
     */
    public AjaxResult(int code, String msg, Object data) {
        super.put(CODE_TAG, code);
        super.put(MSG_TAG, msg);
//        if (CarcheckStringUtils.isNotNull(data)) {
//            super.put(DATA_TAG, data);
//        }
    }

    /**
     * 返回成功消息
     *
     * @return 成功消息
     */
    public static AjaxResult success() {
        return AjaxResult.success("操作成功");
    }

    /**
     * 返回成功数据
     *
     * @return 成功消息
     */
    public static AjaxResult success(Object data) {
        return AjaxResult.success("操作成功", data);
    }

    /**
     * 返回成功消息
     *
     * @param msg 返回内容
     * @return 成功消息
     */
    public static AjaxResult success(String msg) {
        return AjaxResult.success(msg, null);
    }

    /**
     * 返回成功消息
     *
     * @param msg  返回内容
     * @param data 数据对象
     * @return 成功消息
     */
    public static AjaxResult success(String msg, Object data) {
        return new AjaxResult(HttpStatus.OK.value(), msg, data);
    }


    /**
     * 通过 HttpServletResponse 返回success结果
     *
     */
    public static void success(HttpServletResponse response, String message) throws IOException {
        responseResult(response, success(message));
    }

    /**
     * 通过 HttpServletResponse 返回success结果
     *
     */
    public static void success(HttpServletResponse response) throws IOException {
        responseResult(response, success());
    }

    /**
     * 返回错误消息
     *
     * @return
     */
    public static AjaxResult error() {
        return AjaxResult.error("操作失败");
    }

    /**
     * 返回错误消息
     *
     * @param msg 返回内容
     * @return 警告消息
     */
    public static AjaxResult error(String msg) {
        return AjaxResult.error(msg, null);
    }

    /**
     * 返回错误消息
     *
     * @param msg  返回内容
     * @param data 数据对象
     * @return 警告消息
     */
    public static AjaxResult error(String msg, Object data) {
        LOGGER.error("服务报错：{}", msg);
        if (msg.contains("Exception")) {
            return new AjaxResult(HttpStatus.INTERNAL_SERVER_ERROR.value(), "系统内部错误", data);
        }

        if (msg.contains("script")) {
            return new AjaxResult(HttpStatus.INTERNAL_SERVER_ERROR.value(), "系统参数错误", data);
        }
        return new AjaxResult(HttpStatus.INTERNAL_SERVER_ERROR.value(), msg, data);
    }

    /**
     * 返回错误消息
     *
     * @param code 状态码
     * @param msg  返回内容
     * @return 警告消息
     */
    public static AjaxResult error(int code, String msg) {
        return new AjaxResult(code, msg, null);
    }

    /**
     * 通过 HttpServletResponse 返回error结果
     *
     */
    public static void error(HttpServletResponse response, String message) throws IOException {
        responseResult(response, error(message));
    }

    /**
     * 通过 HttpServletResponse 返回error结果
     *
     */
    public static void error(HttpServletResponse response, int code, String msg) throws IOException {
        responseResult(response, error(code, msg));
    }



    /**
     * 通过 HttpServletResponse 返回结果
     *
     */
    public static void responseResult(HttpServletResponse response, AjaxResult result) throws IOException {
        response.setContentType("application/json;charset=utf-8");
        String responseJson = JSONUtil.toJsonStr(result);//JSON.toJSONString(result);
        response.getWriter().print(responseJson);
    }

    /**
     * 方便链式调用
     *
     * @param key   键
     * @param value 值
     * @return 数据对象
     */
    @Override
    public AjaxResult put(String key, Object value) {
        super.put(key, value);
        return this;
    }
}
